This pull request includes a minor change to the PreTrainedModel class in the src/models.js file. The change improves the robustness of the code by using optional chaining to handle undefined values more gracefully.

• src/models.js: Modified the batch_size calculation to use optional chaining for accessing dims property, ensuring the code handles cases where dims might be undefined.

Thanks for the PR! Can you provide the sample code which produces the error?

Thanks for the PR! Can you provide the sample code which produces the error?

I found this when running the example code

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.